#ifndef RAPIDXML_ITERATORS_HPP_INCLUDED
#define RAPIDXML_ITERATORS_HPP_INCLUDED

// Copyright (C) 2006, 2009 Marcin Kalicinski
// Version 1.13
// Revision $DateTime: 2009/05/13 01:46:17 $
//! \file rapidxml_iterators.hpp This file contains rapidxml iterators

#include "rapidxml.hpp"

namespace rapidxml {

//! Iterator of child nodes of xml_node
template <class Ch>
class node_iterator {

public:
	typedef typename xml_node<Ch> value_type;
	typedef typename xml_node<Ch>& reference;
	typedef typename xml_node<Ch>* pointer;
	typedef std::ptrdiff_t difference_type;
	typedef std::bidirectional_iterator_tag iterator_category;

	node_iterator() : m_node(0) {}

	node_iterator(xml_node<Ch>* node) : m_node(node->first_node()) {}

	reference operator*() const {
		assert(m_node);
		return *m_node;
	}

	pointer operator->() const {
		assert(m_node);
		return m_node;
	}

	node_iterator& operator++() {
		assert(m_node);
		m_node = m_node->next_sibling();
		return *this;
	}

	node_iterator operator++(int) {
		node_iterator tmp = *this;
		++this;
		return tmp;
	}

	node_iterator& operator--() {
		assert(m_node && m_node->previous_sibling());
		m_node = m_node->previous_sibling();
		return *this;
	}

	node_iterator operator--(int) {
		node_iterator tmp = *this;
		++this;
		return tmp;
	}

	bool operator==(const node_iterator<Ch>& rhs) { return m_node == rhs.m_node; }

	bool operator!=(const node_iterator<Ch>& rhs) { return m_node != rhs.m_node; }

private:
	xml_node<Ch>* m_node;
};

//! Iterator of child attributes of xml_node
template <class Ch>
class attribute_iterator {

public:
	typedef typename xml_attribute<Ch> value_type;
	typedef typename xml_attribute<Ch>& reference;
	typedef typename xml_attribute<Ch>* pointer;
	typedef std::ptrdiff_t difference_type;
	typedef std::bidirectional_iterator_tag iterator_category;

	attribute_iterator() : m_attribute(0) {}

	attribute_iterator(xml_node<Ch>* node) : m_attribute(node->first_attribute()) {}

	reference operator*() const {
		assert(m_attribute);
		return *m_attribute;
	}

	pointer operator->() const {
		assert(m_attribute);
		return m_attribute;
	}

	attribute_iterator& operator++() {
		assert(m_attribute);
		m_attribute = m_attribute->next_attribute();
		return *this;
	}

	attribute_iterator operator++(int) {
		attribute_iterator tmp = *this;
		++this;
		return tmp;
	}

	attribute_iterator& operator--() {
		assert(m_attribute && m_attribute->previous_attribute());
		m_attribute = m_attribute->previous_attribute();
		return *this;
	}

	attribute_iterator operator--(int) {
		attribute_iterator tmp = *this;
		++this;
		return tmp;
	}

	bool operator==(const attribute_iterator<Ch>& rhs) { return m_attribute == rhs.m_attribute; }

	bool operator!=(const attribute_iterator<Ch>& rhs) { return m_attribute != rhs.m_attribute; }

private:
	xml_attribute<Ch>* m_attribute;
};

} // namespace rapidxml

#endif
